<?php
// +----------------------------------------------------------------------
// | ThinkPHP [ WE CAN DO IT JUST THINK ]
// +----------------------------------------------------------------------
// | Copyright (c) 2006~2019 http://thinkphp.cn All rights reserved.
// +----------------------------------------------------------------------
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
// +----------------------------------------------------------------------
// | Author: liu21st <liu21st@gmail.com>
// +----------------------------------------------------------------------
declare(strict_types=1);

namespace app\common\middleware;

use Closure;
use think\App;
use think\Config;
use think\event\LogWrite;
use think\Response;
use think\response\Redirect;
use think\facade\Request;

/**
 * 页面Trace中间件
 */
class JsonTraceDebug
{
    static $IS_TRACE = false;

    /**
     * Trace日志
     * @var array
     */
    protected $log = [];

    /**
     * 配置参数
     * @var array
     */
    // protected $config = [];

    protected $config = [
        'tabs' => ['base' => '基本', 'file' => '文件', 'info' => '流程', 'notice|error' => '错误', 'sql' => 'SQL', 'debug|log' => '调试'],
    ];

    /** @var App */
    protected $app;

    public function __construct(App $app, Config $config)
    {
        $this->config = array_merge_recursive($this->config, $config->get());
        $this->app    = $app;

    }

    /**
     * 页面Trace调试
     * @access public
     * @param Request $request
     * @param Closure $next
     * @return void
     */
    public function handle($request, Closure $next)
    {
        if (! $this->config['app']['json_trace']) {
            return $next($request);
        }

        $username = $this->config['app']['json_trace_user'];
        // dump(session('admin.username') , $username);die;
        if (session('admin.username') != $username) {
            return $next($request);
        } else {

        }

        $debug = $this->app->isDebug();

        // 注册日志监听
        if ($debug) {
            $response = $next($request);

            if (! $this->isAjaxOrJsonRequest()) {

                return $response;

            } elseif ($response->getCode() == 204) {
                return $response;
            }
            try {
                //code...

                $data = $response->getContent();
                $data = $this->isValidJson($data);
                if (! $data) {
                    return $response;
                }
                $this->log = [];
                $this->app->event->listen(LogWrite::class, function ($event) {
                    if (empty($this->config['channel']) || $this->config['channel'] == $event->channel) {
                        $this->log = array_merge_recursive($this->log, $event->log);
                    }
                });

                if ($response instanceof Redirect) {
                    //TODO 记录
                } else {
                    $log = $this->app->log->getLog('');
                    $log = array_merge_recursive($this->log, $log);

                    $data['trace'] = $this->output($log);

                }
                $response->content(json_encode($data));
            }
            catch (\Throwable $th) {
                return $response;
            }
        } else {
            $response = $next($request);
        }

        return $response;
    }

    /**
     * 判断是否为 AJAX 或 JSON 请求
     * @return bool
     */
    private function isAjaxOrJsonRequest(): bool
    {
        $contentType    = Request::header('content-type');
        $accept         = Request::header('accept');
        $xRequestedWith = Request::header('x-requested-with');

        return (
            $xRequestedWith === 'XMLHttpRequest' || // AJAX 请求
            stripos($contentType, 'application/json') !== false || // JSON 请求
            stripos($accept, 'application/json') !== false // JSON 请求
        );
    }
    function isValidJson($string)
    {
        // 尝试解码字符串
        $decoded = json_decode($string, true);

        // 检查是否成功解码且没有错误
        return (json_last_error() === JSON_ERROR_NONE) ? $decoded : false;
    }
    /**
     * 调试输出接口
     * @access public
     * @param  Response  $response Response对象
     * @param  array     $log 日志信息
     * @return string|bool
     */
    public function output(array $log = [])
    {

        $request = $this->app->request;

        // 获取基本信息
        $runtime = number_format(microtime(true) - $this->app->getBeginTime(), 10, '.', '');
        $reqs    = $runtime > 0 ? number_format(1 / $runtime, 2) : '∞';
        $mem     = number_format((memory_get_usage() - $this->app->getBeginMem()) / 1024, 2);

        if ($request->host()) {
            $uri = $request->protocol().' '.$request->method().' : '.$request->url(true);
        } else {
            $uri = 'cmd:'.implode(' ', $_SERVER['argv']);
        }

        // 页面Trace信息
        $base = [
            '请求信息' => date('Y-m-d H:i:s', $request->time() ?: time()).' '.$uri,
            '运行时间' => number_format((float) $runtime, 6).'s [ 吞吐率：'.$reqs.'req/s ] 内存消耗：'.$mem.'kb 文件加载：'.count(get_included_files()),
            '查询信息' => $this->app->db->getQueryTimes().' queries',
            '缓存信息' => $this->app->cache->getReadTimes().' reads,'.$this->app->cache->getWriteTimes().' writes',
        ];

        if (isset($this->app->session)) {
            $base['会话信息'] = 'SESSION_ID='.$this->app->session->getId();
        }

        $info = $this->getFileInfo();

        // 页面Trace信息
        $trace = [];
        foreach ($this->config['tabs'] as $name => $title) {
            $name = strtolower($name);
            switch ($name) {
                case 'base': // 基本信息
                    $trace[$title] = $base;
                    break;
                case 'file': // 文件信息
                    $trace[$title] = $info;
                    break;
                default: // 调试信息
                    if (strpos($name, '|')) {
                        // 多组信息
                        $names  = explode('|', $name);
                        $result = [];
                        foreach ($names as $item) {
                            $result = array_merge($result, $log[$item] ?? []);
                        }
                        $trace[$title] = $result;
                    } else {
                        $trace[$title] = $log[$name] ?? '';
                    }
            }
        }

        if (isset($trace['SQL']) && is_array($trace['SQL'])) {
            $trace['SQL'] = array_reverse($trace['SQL']);

        }
        return $trace;
    }


    /**
     * 获取文件加载信息
     * @access protected
     * @return integer|array
     */
    protected function getFileInfo()
    {
        $files = get_included_files();
        $info  = [];

        foreach ($files as $key => $file) {
            $info[] = $file.' ( '.number_format(filesize($file) / 1024, 2).' KB )';
        }

        return $info;
    }
}
